package cn.edu.tju.wallet.repository;

import cn.edu.tju.wallet.domain.VirtualWallet;
import jakarta.persistence.LockModeType;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Lock;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.util.Optional;

/**
 * 虚拟钱包Repository接口
 */
@Repository
public interface VirtualWalletRepository extends JpaRepository<VirtualWallet, Long> {

    /**
     * 根据用户ID查询钱包
     * @param userId 用户ID
     * @return 钱包对象
     */
    Optional<VirtualWallet> findByUserId(Long userId);

    /**
     * 根据用户ID查询钱包（带悲观锁）
     * 用于支付等需要并发控制的场景
     * @param userId 用户ID
     * @return 钱包对象
     */
    @Lock(LockModeType.PESSIMISTIC_WRITE)
    @Query("SELECT w FROM VirtualWallet w WHERE w.userId = :userId")
    Optional<VirtualWallet> findByUserIdWithLock(@Param("userId") Long userId);

    /**
     * 根据ID查询钱包（带悲观锁）
     * 用于支付等需要并发控制的场景
     * @param id 钱包ID
     * @return 钱包对象
     */
    @Lock(LockModeType.PESSIMISTIC_WRITE)
    @Query("SELECT w FROM VirtualWallet w WHERE w.id = :id")
    Optional<VirtualWallet> findByIdWithLock(@Param("id") Long id);

    /**
     * 检查用户是否已有钱包
     * @param userId 用户ID
     * @return 是否存在
     */
    boolean existsByUserId(Long userId);
}

